package cn.zqy.demo2.day11Net.OptimizeFileUpload;


import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Random;

/**
 *
 * 文件上传案例服务器端:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写"上传成功"
 *
 *     明确:
 *         数据源:客户端上传的文件
 *         目的地:服务器的硬盘 e:\\itma\\1.jpg
 *
 *     实现步骤:
 *         1.创建一个服务器ServerSocket对象,和系统要指定的端口号
 *         2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
 *         3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
 *         4.判断e:\\itma文件夹是否存在,不存在则创建
 *         5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
 *         6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
 *         7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
 *         8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
 *         9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
 *         10.释放资源(FileOutputStream,Socket,ServerSocket)
 * */

public class Demo02Server {
    public static void main(String[] args) throws IOException {
        // 1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(8888);

        /*
            让服务器一直处于监听状态(死循环accept方法)
            有一个客户端上传文件,就保存一个文件
         */

        while (true) {

            // 2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
            // 必须把这部分放在线程任务外面
            Socket socket = server.accept();
            new Thread(new Runnable() {
                @Override
                /*
                使用多线程技术,提高程序的效率
                有一个客户端上传文件,就开启一个线程,完成文件的上传
             */
                public void run() {
                    try{

                        // 3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = socket.getInputStream();
                        // 8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
                        OutputStream os = socket.getOutputStream();

                        //4.判断e:\\itma文件夹是否存在,不存在则创建
                        File file = new File("e:\\itma");
                        if (!file.exists()) {
                            file.mkdir();
                        }
                        // 5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        String fileName = "img-" + System.currentTimeMillis() + "_" +
                                new Random().nextInt(1000) + ".jpg";
                        FileOutputStream fos = new FileOutputStream(file + File.separator + fileName);

                        int len = 0;
                        byte[] b = new byte[1024];
                        // 6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
                        while ((len = is.read(b)) != -1) {
                            //7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
                            fos.write(b,0,len);
                        }

                        //9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
                        os.write("收到谢谢！".getBytes());

                        //10.释放资源
                        fos.close();
                        socket.close();
                    } catch (IOException e) {
                        System.out.println(e.getMessage());
                    }
                }
            }).start();

        }
        //服务器就不用关闭,一直处于监听状态
    }
}
